#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Time    : 2024/1/22 14:11
# @Author  : wanghaoran
# @File    : __init__.py.py
from reranker.inference import RerankerForInference
from configs import RERANK_MODEL_PATH, device

# reranker_model, tokenizer = RerankerForInference(RERANK_MODEL_PATH).from_pretrained()  # load checkpoint
reranker_model = RerankerForInference.from_pretrained(RERANK_MODEL_PATH).to(device)  # load checkpoint
